package com.docchatlink.iflytek2.vo;

import lombok.Data;

import java.util.List;

/**
 * <p>
 * 文档内容相似度检索VO
 * </p>
 *
 * @author Mr.piggy
 * @since 2024/6/24
 */
@Data
public class VectorSearchParam {
    /**
     * 文件id列表，不传，查app下所有文件
     */
    private List<String> fileIds;
    /**
     * 向量库查询数量
     * 默认：5
     */
    private int topN;
    /**
     * 用户的问题
     */
    private String content;

    private ChatExtends chatExtends;

    @Data
    public static class ChatExtends {
        /**
         * WIKI结果分数阈值，低于这个阈值的结果丢弃
         * 默认：0.82
         */
        private Float wikiFilterScore;
        /**
         * 进行全文检索时的es分数阈值，低于这个阈值的结果丢弃。
         * 默认：10
         */
        private Float esFilterScore;
    }
}
